Working on perfecting height calculation