I think before you try to resolve this silly little "problem" you're having you should first understand the code you're trying to modify. Try to rebuild the Movement system in your own project, for two reasons:

1.) You'll know exactly how it works because you made it and won't have to ask everyone else for help every time you come to a problem with modifying it.

2.) The given Movement example is coded rather poorly not to mention slightly outdated functions/variables used in it.