a small game of python