Corrected mistake wit class naming