That works! I was a little paranoid about making things unpickleable so I was trying to build this out of more primitive python objects.