Simplify lib.rs