College project in Python