improved AI a little bit in the right direction