problem_003.py : Added hints to make better

Added certain hints to make the code better.

Don't worry about rewriting the complete program even if it is working. That is one of the things about software, you have to quite commonly rewrite complete programs or functions.