ENH: Better visualization. Add script