Minor revision to hw2/hw3