Update test with new schema, small pep8
Add some documentation