After modifying the code for benefits