Neaten up code.