An improved parton shower algorithm in QED