Added menu, added AI again, improved loop for smooth animation