Dynamical heterogeneities and mechanical non-linearities: Modeling the onset of plasticity in polymer in the glass transition