added code to readme