Clean up
Add AI
Readd AI