[TASK] Update README.md and some style cleanup