Cleaned up AI interface