Add Tests

[anuraghazra]

Adding tests would be useful in this case and also will be easier to refactor code.

We could use jest for testing.

[sdkdeepa]

Hello! I would like to contribute to this.