Add DBException in models instead of ifs