Improving Generalization for Abstract Reasoning Tasks Using Disentangled Feature Representations