Add logging using logging module