ENH: Improved sample script