Refactored to reflect changes to the strategy
Refactored to reflect changes made to the strategy
Updated assert method
Refactored to reflect changes made to the strategy