Add slightly smarter AI