AI Updates

Improved AI functionality
Refactored menus