Add a test that the custom class name is added