root directory style cleanup

Some end of the day style cleanup :alarm_clock: 

Root directory passes flake8