Yes, but be sure to leave the function signature exactly as it is provided. Otherwise, our assessment engine will not be able to run your program. With that in mind, feel free to add helper functions, imports/includes, etc. Be aware that some tasks feature additional restrictions on how much of the initial code you are allowed to modify. In these cases, the task is evaluating your skill in making as few edits as possible to address the stated problem.
Can I modify the initial code?
