Small modifications to lab11