Visualization improved, but still some bugs in the wheel callback