experimenting with numpy