Fix style issues for PEP8. Add key.py