Pep8fy
Add support to test multiple graphs - still need to improve test coverage, remove code duplicated and update docs