tidy up
code tidy