Refactored visualization widget with more options and controls