Update rails generator code