A robot game in python