Updates for lab 2