update generation script rake task