put numpy part back in