Some changes :) Plots are beautiful!!!