starting to work on the improved ai