cleanup with help of pylint