add the patch from TAs in Lab 2.3