Update the given Python program to shuffle an array of numbers.
[1, 2, 3, 4, 5, 6]
import random

def shuffleArray(arr):
    random.shuffle(arr)
    return arr

array = [1, 2, 3, 4, 5, 6]
shuffledArray = shuffleArray(array)
print(shuffledArray)