minor updates but code coverage is working in this example