Add integrational tests