Small corrections to script