Python solution for the arithmetic equation challange [EASY].
Creating a new function to do the processing after receiving the input.