Using arrays of numpy