Tidy up
Code clean