Updated setup script for running on AWS headnode