[Feature request] allow comments on code snippets in a code block

It seems we currently cannot add comment to specific code in a code block. It will be great to support selecting specific lines (eg. numpy) and add comments accordingly

```
import pandas as pd
import numpy as np
```