Diego Costa has taken his first step towards reintegration into the Chelsea squad following his spat with Antonio Conte