University project in Python