I've got this working on my live site and it's running fine.

Thanks for this bit of code, it's very useful. :)