Wrote basic test suite