Update to readme and class file