Fix movement and AI