Update readme and add code