Update the save routine for submenu context

Resolve #139
Copypasta I think