Tidy up andbudget.py