Added explicit import numpy and matplotlib