Update for lab4