My solution to Option 2