PEP8 fixes.
Regex fix.
Removed unneeded import