Add some unit tests to validate the functionality. See [this comment](https://github.com/codefornola/211-data-wrangling/pull/1#pullrequestreview-404027433) for some ideas on how to do this.