* Update `feature_visualization()`

Only plot for data with height, width > 1

* cleanup

* Cleanup