this is my assignment in python