Added code to the readme