Add write_checkpoint option to input

Add this option to allow for reading of function space information for post-processing.

• Implement this feature
• Document and test it