Added docstrings to heightmap.py