Add Class in Main and modify README