python program using if ..elif else statement