Add logging statements