made quick changes to run on cluster