Assignment in Python, CS-242