File modified to get input data ready for plotting graphs and also for demo purpose