Add readme
class refactoring