[TESB-13771] - Allow specify custom class as argument
Merge pull request #13 from i-melnik/master

[TESB-13771] - Allow specify custom class as argument