Added code to save before and after context vectors