Added integration tests for distribution