python programs for math