Fixed an indentation and added some pickling.