Improve using node prob display