Oo good idea! I have some other improvements in mind as well, so I think I'll give it another patch when I get the chance. Thanks!
