Make sample size configurable