Different school assignments in python