Set up code documentation

Set up code documentation in order to easily understand the python packages without having to dig through the source code.