UI Tweak & bug fix