Need to add a `done` parameter here, or the function ends immediately after running the last line of the code.