adding wrap samples in src/utils.py