Tidy up local simulation script