small changes to setup and run