Update build-a-k-nearest-neighbor-model-with-p5js.mdx

Author: sonnynomnom

projects/build-a-k-nearest-neighbor-model-with-p5js/build-a-k-nearest-neighbor-model-with-p5js.mdx

@@ -228,17 +228,21 @@ ellipse(mouseX, mouseY, 10, 10);
 
 Now we just need to call it! At the bottom of the `draw()` function, call `classifyMouse()`. Feel free to move your mouse around and see if the classification is correct! Is your mouse pointer red when the majority of the 3 nearest neighbors are also red?
 
-## Extensions:
+## Congrats!
 
-You made a working KNN visualizer! Want to level it up? Try these challenges:
+Congrats on reaching the end of this project! You made a working KNN visualizer! 
+
+You now hopefully have a better understanding of how a K-Nearest Neighbor model works under the hood!
+
+### Bonus Challenges
+
+Want to level it up? Try these bonus challenges:
 
 - We've hard-coded `20` points in our dataset and `3` nearest neighbors. Refactor your code so it is easy to quickly change those values to any numbers.
 - It can be a bit tough to eyeball which points are the nearest neighbors to your mouse. Edit your code so the nearest neighbors are now highlighted (maybe with a yellow border rather than black)
 - Add your own style! Rather than red and green points, you could draw your points as wizard 🧙‍♂️ or warrior ⚔️ emojis (calling back to our original example). Or come up with your own theme! Could you do something interesting with the background?
 - KNN models can work with multi-class problems. Rather than simply red and green points, what if you had red, green, and blue points? Refactor your code so your visualization works with _any_ number of classes.
 
-Congrats on reaching the end of this project! You now hopefully have a better understanding of how a K-Nearest Neighbor model works under the hood!
-
 ### Additional Resources
 
 - [p5.js](https://p5js.org)