update scala code