Fix tests.
pep8.
Remove init_db
Fix test