Python Arithmetic classwork