Updates to assertWillThrow method