Update readme
Add main code