Allow writing of arbitrary fields for post-processing

[ia267]

Made write_fields subroutine general so that it'll work with any field.

At the moment checkpointing itself still uses a fixed output (u,v,w). I'll open another issue to handle arbitrary fields during postprocess procedure.

Originally posted by @ia267 in #173 (comment)