Include test files on distribution
Improve class name