fifth week python assignment working code