Updated with suggestion from review.