Made a few tweaks and improved on the documentation