Improve Staging Extension UI