Poor compatibility with Jupyter Notebooks

[rohitsathish]

Hi, This is my favourite them on VSCode but unfortunately, some changes have caused it to appear poorly in Juyter Notebooks. It used to be similar to the regular python version before though. Would appreciate if there are any simple tweaks that can get a better output.