refactor the lib source file

For modularity, I think we should separate different features into their own module and retrieve them in the `lib.rs`.

That way, if anything breaks on one module, it can be isolated and fixed
At the same time, it will become easy to write smaller unit tests that have few dependencies