Wrote some unit tests and they pass