Creating a patch

This is a patch for the code.
Adding a test class

Testing
Merge pull request #2 from brentonsnhu/Bugfixes

Bugfixes