Working on numpy munging script