Tidy up of heights changed method