Add more documentation

• Adds documentation to all the public facing source code that was not documented earlier.
• Adds #![warn(missing_docs)] to the top of src/lib.rs which generates warnings if any public code is undocumented.
• Adds more diagrams