add code to run on sample code