Working on unit tests for Device class