Made following changes:
1] Added docstring tests
2] Fixed test failures