Python for math