Added code to readme