Add setCurrentState to Device class
Updated Model tests