1. Move `ImagePanel` class to the visualization module. 2. Remove duplication with the classes in `PlotConv2D.kt`. 3. Extract the common code which creates a frame to a function.