Small fixes in sample.