Add logging for models