Add changes to lab5