Clean up UI slightly.