Working on square movement