This version is working with sample data

I just need to fix an small problem regarding to plotting