Fixed code in the readme