Added HW after fixing git issues.

2 files added.
Per the assignment, there should be an `if __name__ == '__main__'` block that demonstrates your code works.