Add a class parameter
Merge pull request #8 from mremy/add_class_param_on_base

Add a class parameter