Fixes issue #688 with slightly inadequate geological map as a result of plot_2d()

[mkondratyev85]

Description

Please include a summary of the changes.

Relates to issue #688

Checklist

• My code follows the PEP 8 style guidelines.
• My code uses type hinting for function and method arguments and return values.
• My code contains descriptive and helpful docstrings
  which are formatted per the Google Python Style Guidelines.
• I have created tests which entirely cover my code.
• The test code either 1. demonstrates at least one valuable use case (e.g. integration tests)
  or 2. verifies that outputs are as expected for given inputs (e.g. unit tests).
• New and existing tests pass locally with my changes.

[mkondratyev85]

After fixing this issue you can see that the resulting geological map corresponds to what we actually see on cross sections (as compared to what we have seen in #688 )

[Japhiolite]

could've also done with order='F' in reshape, but transposing is fine too