Tidy up script
Tidy up script