Extended the code styling to the corrector stage tool