Cleanup
More assertions in sample.