Changing functionality of withMessage method to accept the full instance.