numpy assignment  finished