Updated pull request for HW4

Pull request after the changes made.

Thank you for letting me do this.
-Ari