Money budgeting in python