Added learning AI and difficulty selector