Minor tweak to window flatmap with state.

Slight performance improvement for the case where we are fast forwarding over many loci ranges.

Also fix a code formatting issue.