Add propper logging

 * Explicitly import logging due to possible graph reordering