DB update bug fix + UI update