Made it look better
hw4 update
Forgot to add templates
Merge pull request #7 from Karmeow/hw4

hw4