Input can now be a vector