adding some plotting