Tidy up

Tidy up