Adding integration tests for PR #8

See #8