Changes following review

Signed-off-by: Jake Smith <jake.smith@lexisnexis.com>