More work on lab 2