added correct code for stratigy pattern