Add integrated tests