# Clean and prepare data
df['bmi'] = df['weight'] / (df['height']/100)**2  # Create new feature
df = df.dropna()  # Remove missing values

# Separate features and target
X = df[['height', 'age']]  # Features
y = df['weight']           # Target