Add exception checking