Fix README after refactor