Updated for Yun

Updated code to work on YUN