Update generator and Gemspec to match modified register method