Ben's code

Got it. Thanks, Prof. Ben.