# add in squares of mother and father heights
male_df['father_sqr'] = male_df['father'] **2
male_df['mother_sqr'] = male_df['mother'] **2