Added extra props for visualization