Arithmetic Function In Python