Cleanup unused parameters

Changelog:
- Cleaning up unused parameters
Instead of dropping the argument here, maybe you should pass it over to the `cb`. It looks like it's expecting it:
```python
def cb(msg, handshake_cb=handshake_cb):
#...
```