Fixed comparision of model and real data. Moved this to show comparision after optimization.