Wrote unit tests