fixing propagator errors, improving plotting