Problem 1 Solution w/ Vector