modifications to lab 2