Updated docs to match new file structure and module declaration