Adding onHeightChange callback