added bit more to script