(very) small changes to script