Adding Strategy pattern in python