Improve UT commenting