modify code style of problem_3 & 4 to satisfy PEP 8 coding style