Adding few integration tests