cosmetic comment: please put a whitespace after the `#` sign when used for inline comments:

```python
    # w_norm2 = np.dot(w, w)
```