Add logging to controllers