fix word-count module/import for test runner

Also refactors assignment implementation.