Done and also made the plots look better.