Update assignment 4 height :art: