Adds missing end to class