Updating code to run on atomcool.