Fixed.
Merge pull request #1 from jgillham/master

Update your code with my code