Update stage4_submission.py

updated function name to stage4