Some budget stuff in Python