diff --git a/ml_bootcamp/Regression_Exercise.ipynb b/ml_bootcamp/Regression_Exercise.ipynb
index 73e6324d954db4aa9e3d7f3bac6ec691bb20e222..1e26d265c0cb322734da58e1ee88c97440bf659a 100644
--- a/ml_bootcamp/Regression_Exercise.ipynb
+++ b/ml_bootcamp/Regression_Exercise.ipynb
@@ -280,7 +280,7 @@
     "missing_data.head(20)\n",
     "\n",
     "f, ax = plt.subplots()\n",
-    "plt.xticks(rotation='90')\n",
+    "plt.xticks(rotation='vertical')\n",
     "sns.barplot(x=X_train_na.index, y=X_train_na)\n",
     "plt.xlabel('Features', fontsize=15)\n",
     "plt.ylabel('Percent of missing values', fontsize=15)\n",
@@ -295,7 +295,7 @@
    "outputs": [],
    "source": [
     "## visualize correlation\n",
-    "sns.heatmap(X_train.corr())\n",
+    "sns.heatmap(X_train.select_dtypes(np.number).corr())\n",
     "plt.show()"
    ]
   },