Python program to create and modify a vector