Added the shader code for moving and refactored some code