Budget for a construction company in Python