Added back plotting capability.