Update readme and comment