Improve volume node box