Updated spark on yarn code