Add check for array