Add function to calculate pyramid height from blocks

[akoalp]

Describe your changes

Checklist

• I have read the [CONTRIBUTING]
• I have performed a self-review of my own code
• I have run the code locally and it works as expected
• I have commented my code, particularly in hard-to-understand areas

Screenshots (if appropriate)