Added some plotting functions