Your question interested me, and I found an interesting solution here.

It's a bit more complicated than your code, but will work. All you need to do is change this line

to this