equal numbers program in python