Added code snippet to readme