Basic Math Formulas Geeksforgeeks Python