Modifying the pom-pom model for extensional viscosity overshoots