how to modify a$4.$py code to pass output testcase

pic$1$ is requirment

pic$2$ is a$4.$py

pic$3$ is all code file

pic$4$ is expected output