Add JMX support