added more tests for integration