save wm trials, move save trial code into own function/file