Update README
Change class to module