Fixing #6 and writing tests for DeviceDriver handshake