Added addState method to Device class
Updated tests for devices