Add unit testing and integration tests
Add unit testing and integration tests