Added support to specify class name manually instead of getting it out of model.