Minor change for hw4 part 2 -- no time now to prettify