Post-generation tweaks