Just some tidying of the code