Update class name in design doc and better formatting for class names.