Added integration tests for implemented functionality