Add part of Integration testing