made function for budget

FUNKTIONZZZ!!!