Messing around a bit more with numpy