update code for new probe design