Change class to public