Add NaN check