Improve movement algorithm to allow for taking the cube (currently doesn't work)