Fixed budget bugs and added progress bar