Add code and write the usage in README.md