import numpy as np