Polishing Chapter 5 code and README
Update the README