Small error corrected for imports + a bit of Pylint