Corrected class declaration