Added import to numpy

Signed-off-by: Renato Stoffalette Joao <renatosjoao@gmail.com>