cleaning up some tests and adding type hints
Merge pull request #14 from paulbellamy/performance

cleaning up some tests and adding type hints