Ah, good call. Updated the code and added a new commit below.