Code tidying
Code tidy