Slight improvment of the plots