Replace Conditional with Polymophism