Update UI & bug fix