Fixed the docs and formatting for the new implementation