Just pushed a quick change to this--if it still needs work let me know.