Minor GUI fixing