Fixed some errors in 4th lab