Added import numpy statement.