Allow arrays + vectors; impute vectors for some plots