fixed the plots a little bit