Simple drinkinggame written in Python