Add tests - v2