GUI update & bug fix