Organized includes import
Fixed entropy calculation
Updated visualization module