University cw writtin in Python for COM506