Fix mock result
Fix imports (tests)
pep8/pylint
Add test for multiple results
Code cleanup