Stress relaxation in diblock copolymers