Outline of problem3.py including inputs and query