tidy up the script a bit