how do i write a program in python for an atm machine