fix setup_database method
save budget in database