I think this is already implemented in <code>Budget.getBudgetById(budgetId)</code>