Fixing NaN bug in columns