Add total height calculation, but animiation still spoilt.