trying to improve the gui a little