Fix code in readme