Add ai and player classes