fix yo bugs brah
Merge pull request #2 from chintanparikh/master

fix yo bugs brah