Removed the input of max_atoms and max_heavy_atoms