python assignment solutions