Incorporate sampling step