forked original gem and then updated with my code