Update readme and add the code