Add sample code to readme