Add budget to team model