Wrote basic unit tests