fixes after debugging
remove numpy import