Modified script to include test coverage.

@YolinaS
@gtrogers