A simple AI coded in python