Update readme
Put the code in the right place