import matplotlib.pyplot as plt
import numpy as np
import os
from IPython import embed