ENH: address discussion + COSMITS + fix viz