Updated use statement to new module