(refactor)Small edits to src/lib.rs