Wrote test code