Updating failing tests for #36
Merge pull request #37 from iamkhush/master

Updating failing tests for #36