Refactor code, move distributed implementation into a submodule