Small edits in GUI