Cleaned up in single player AI