Working on adding AI

Working on adding AI