Add some more GUI logic.