Adding sample code to test