Add tests framework