My first mini 'game' in Python