Update tests.
Fix PEP8.