Wrote unit tests for all the things