add math code & fix me errors