Added some plotting stuff