Which of the two sentences shows correct parallel structures?

A. Professor Aslam rewarded his students for working diligently in the final test and going beyond the call of duty
B. Professor Aslam rewarded his students for their diligent work in the final test and going beyond the call of duty.