Made the code more pythonic#7

noviicee wants to merge 1 commit intoclassroom-resources:masterclassroom-resources/autograding-example-python:masterfrom

noviicee:issue-2noviicee/autograding-example-python:issue-2Copy head branch name to clipboard