lab update

lab 5.1 works now
updated code

some changes needed to be made