Organize imports
Make signature full-capital since it is a constant
Fix PEP8 errors