minor edits to hw2 prob 1 and 2