added a nice plotting option