small fix
added device class structure