python atm code using if else