adding tests

I want to add tests first