Found a better modification to make