Test fixes, PEP8 fixes