Rewrite class comment