Minor GUI improvement