GUI

Little GUI improvement