Add logging to lab2