Updated use statement so module passes unit tests