added a variable for the budget and added an if else block to test code