Updated the whole gem