add gui improvements