fix mistakes in Lab4