A simple cash register program in Python