Add logging decrator