review of hw1

Added a few suggestions to make the code easier to expand on.