google-nn script modifications