Code optimisation (thanks @mbabker)