Update custom_fuction.r