Added class with correct name