diff --git a/source/tm/team_predictor/client/client.js b/source/tm/team_predictor/client/client.js
index 6025f1e3..61cb141c 100644
--- a/source/tm/team_predictor/client/client.js
+++ b/source/tm/team_predictor/client/client.js
@@ -168,22 +168,30 @@ add_pokemon_button.addEventListener('click', () => {
 	}
 });
 
-document.getElementById('generate').addEventListener('click', send_data);
+document.getElementById('generate-random').addEventListener('click', send_random_data);
 document.addEventListener('keydown', (event) => {
 	if (event.key === 'Enter') {
 		// Without this, the data is submitted twice
 		event.preventDefault();
-		send_data();
+		send_random_data();
 	}
 });
+document.getElementById('generate-most-likely').addEventListener('click', send_most_likely_data);
 
-function send_data() {
-	socket.send(JSON.stringify(team_to_json()));
+function send_data(style) {
+	socket.send(JSON.stringify(team_to_json(style)));
 }
 
-function team_to_json() {
+function send_random_data() {
+	send_data('random');
+}
+
+function send_most_likely_data() {
+	send_data('most likely');
+}
+
+function team_to_json(style) {
 	const generation = document.getElementById('generation').value;
-	const style = document.getElementById('style').value;
 	const team = [];
 	const pokemon_elements = document.querySelectorAll('.pokemon');
 
diff --git a/source/tm/team_predictor/client/index.html b/source/tm/team_predictor/client/index.html
index af0cb229..5ca0f050 100644
--- a/source/tm/team_predictor/client/index.html
+++ b/source/tm/team_predictor/client/index.html
@@ -21,16 +21,13 @@
 	<div class="container">
 		<div class="inputs">
 			<select id="generation"></select>
-			<select id="style">
-				<option value="most likely" selected>Most Likely</option>
-				<option value="random">Random</option>
-			</select>
 			<div id="team"></div>
 			<div id="add-pokemon-container">
 				<button id="add-pokemon">Add Pokémon</button>
 			</div>
 			<div id="generate-container">
-				<button id="generate">Generate team</button>
+				<button id="generate-random">Generate random team</button>
+				<button id="generate-most-likely">Regenerate most likely team</button>
 			</div>
 		</div>
 		<pre id="predicted" class="predicted"></pre>