Attempt to add docstring functionality (Kind of works now)