Update readme
Refactor env setup