Add tests

Would be nice to add a test, at least for the logic part of the device