tidy up code formatting