Add log wrapper