Code with better movement