Fixed bug in Lab2