Fixing small bugs in GUI display