Added directory argument to setup script