change running code to match example