Python program set 4