Refactoring
Changed '' class to ''