Changes from review

Signed-off-by: jamienoss <james.noss@lexisnexis.com>