Splitting Z&S polynomial modifications to separate subroutine. Also moving rescaling to thend of timestep
Fixed some memory allocation issues in coeff_update and main program.