Add tests basic implementation