refactor the lib source file

[shaddyshad]

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

[shaddyshad]

I will make a PR with my suggestions