Update readme and added some comments