[Minor] Class comments updated