1 files changed, 8 insertions, 3 deletions
diff --git a/lab_control/jds6600.py b/lab_control/jds6600.py
index 7e7640d..6137105 100644
--- a/lab_control/jds6600.py
+++ b/lab_control/jds6600.py
@@ -20,18 +20,23 @@ class JDS6600(FunctionGenerator):
     def closePort(self) -> None:
         self._port.close()
 
-    def _queryOnOff(self) -> list[int, int]:
+    def _queryOnOff(self) -> list[str, str]:
+        # TODO double check query on the device
         response = self._sendRequest("r20", "")
         return [response[5], response[7]]
 
     def setOn(self, channel: int) -> None:
         state = self._queryOnOff()
-        state[channel - 1] = '1'
+        state[channel - 1] = "1"
         response = self._sendRequest("w20", f"{state[0]},{state[1]}")
         # TODO figure out error handling
 
     def setOff(self, channel: int) -> None:
         state = self._queryOnOff()
-        state[channel - 1] = '0'
+        state[channel - 1] = "0"
         response = self._sendRequest("w20", f"{state[0]},{state[1]}")
         # TODO figure out error handling
+
+    def setFrequency(self, channel: int, frequency: float) -> None:
+        opcode = f"w{channel + 23}"
+