I worked it out on a second look. Something about that code was confusing, but I think I can add it now :)

:beer: