Fix class in readme