Add math code