hw2 for 11-791.