add support for new final budget allocation