Add the whole gem code