Improvement demo UT