Update R file for assignement