Use find_packages
Add tests for BERT serializer
Rename helpers