bug fix + UI