saving slight tweaks to script