updated to the code on the cluster