Adds in more use of AI_Control. The square controls are bugged slightly.