Roster Planning program in Python