Feature space visualization in explore.py