It's  a little more verbose now, but it still works. I'm sure this can even probably be cleaned up to be less than 4 lines as well, but it works for the time being.