Add integration tests
Merge pull request #4 from paulcox24/Integration_test

Add integration tests