Changes figure to EPS format and added future work