Add lab2 changes

additional changes