Add new comment

Al's picture

I've just updated the code, I'm interested to know if this helps.