Modelling non-adiabatic processes using correlated electron-ion dynamics