Refactor - extract Square and Triangle.