Stress relaxation in polymers: a modified site model analysis