Improved documentation
Fixed bug - decorators were not keeping docstrings