Initial work updating lab 5