removed numpy import - caused an error