Rewrite pomodoro script in Python