Add simple movement handling