Added tests for heights and fixed mistakes found.