add code to readme