Adding movement logic