add heap lab code