Updated sampling script