added code to the readme