Moving force assignment out of key handling function

The actual force assignment isn't done yet. Still need to figure out how
to keep final magnitude constant regardless of direction.