update readme, fixed the code