minor changes to initial script