fixed accidentally removed np import (python version)