Basic version, working on AI