Add exception testing