Small AI fix
Pep8 done!
Coverage 100%