Add math tests