Simplify module import logic
Run the following command from the root of this projcet:
$ python -m unittest discover test
