Improved controls and AI