Clean up

I find myself making these changes to the generated file every time I use this script