Comment by jedi_stannis

My solution is a little bit more hackish, but more usable imo. Not thoroughly tested though, might have some implications I haven't thought of.

http://pastebin.com/m7cc84252