Merge pull request #2 from bharvidixit/patch-1

Update test