Fixing main script
Styling Plots