Update problem.py

submit() seems to need to return the input value.
otherwise the submitResults value is None