integration test part added