Updated the sample for some better testing.