FIX: fixed flake8 warnings