Minor style fixes
Class comments