Faster calcs with numpy