FIX: Use logging module for printing
STY: PEP8
FIX: Fix logging issue
FIX: Fix logging