Wrote unit test