Added support to one more hyper-parameter, the expected number of groups.