Refactor

* Use `test-framework` option
* Code style

Fixes #8.