Added a new flag to control the number of buckets for storing the parameter vector.