Reformat code to use the Budget class

The code now uses the Budget class created before.
This creates a more logical structure of the code.

-Get and set the budget via the Budget class