Create problem_003.py

program does not give desired result