Improve the repo and update README