Fix code in the readme