added some code in gem file