Tidy up script