Small bug fix for hess argument.
Adding tests for new functionality.