Modified class definition for consistency
Added new line as requested