Changed class to public