UpdateHeights refactor

changed function signature