Small GUI improvement