A little bit of house-keeping.

(Made the .py file a little neater)