Finished budget model and tests