Add missing overflow checks