Add code to readme