Hello,
in principle, you could just change the IntegerMutation, but that would also mean, you cannot use it again for future assignments. So it is better to write a new one. You can actually start by copying (and renaming) the old one.
Have a nice day,
Martin