started adding tests, unit test and integration