Tidy up Code