Well, instead of calling it 'having a smoke', it's 'having a toke' (which works either way of course).
But yeah, instead of 'poof', which could raise the eyebrows of people raised in some countries, I'd suggest 'puff'. I think I might have actually used this term myself before.
