Small gui improvement