submitting correction expectation.py