Fixing the budget setup page UI