one return statement removed from highlight fn