Fix class & readme