Revert "Revert "Assignment2 doc checkpoint updated""

This reverts commit f4702a71ee821be710d34ea34ec09815a051a53f.