Added some plotting