Note that we want to work with numpy array and not with list.