diff --git a/machine_learning/neural_network.cpp b/machine_learning/neural_network.cpp
index f77b732bb..978d658b4 100644
--- a/machine_learning/neural_network.cpp
+++ b/machine_learning/neural_network.cpp
@@ -826,7 +826,10 @@ static void test() {
     return;
 }
 
-/** Driver Code */
+/**
+ * @brief Main function
+ * @returns 0 on exit
+ */
 int main() {
     // Testing
     test();