Add customizable height function