Clean up generation script