Good point. I've changed the code and updated the readme. Please review.