diff --git a/logic/models/run.go b/logic/models/run.go
index 8991cd9..0b8b5c4 100644
--- a/logic/models/run.go
+++ b/logic/models/run.go
@@ -118,6 +118,13 @@ func runModelExp(base BasePack, model *BaseModel, def_id string, inputImage *tf.
 }
 
 func ClassifyTask(base BasePack, task Task) (err error) {
+	defer func() {
+		if r := recover(); r != nil {
+			base.GetLogger().Error("Task failed due to", "error", r)
+			task.UpdateStatusLog(base, TASK_FAILED_RUNNING, "Task failed running")
+		}
+	}()
+
 	task.UpdateStatusLog(base, TASK_RUNNING, "Runner running task")
 
 	model, err := GetBaseModel(base.GetDb(), *task.ModelId)
diff --git a/logic/models/train/train.go b/logic/models/train/train.go
index fa0ef6a..7b18af7 100644
--- a/logic/models/train/train.go
+++ b/logic/models/train/train.go
@@ -384,7 +384,6 @@ func trainDefinitionExpandExp(c BasePack, model *BaseModel, definition_id string
 		log.Error("Failed to get the exp head of the model")
 		return
 	} else if len(heads) != 1 {
-		log.Error("This training function can only train one model at the time")
 		err = errors.New("This training function can only train one model at the time")
 		return
 	}
@@ -1450,9 +1449,9 @@ func generateExpandableDefinition(c BasePack, model *BaseModel, target_accuracy
 	// Create the blocks
 	loop := int((math.Log(float64(model.Width)) / math.Log(float64(10))))
 
-	if model.Width < 50 && model.Height < 50 {
-		loop = 0
-	}
+	/*if model.Width < 50 && model.Height < 50 {
+	loop = 0
+	}*/
 
 	log.Info("Size of the simple block", "loop", loop)
 
@@ -1541,7 +1540,7 @@ func generateExpandableDefinitions(c BasePack, model *BaseModel, target_accuracy
 		if model.Width > 100 && model.Height > 100 {
 			generateExpandableDefinition(c, model, target_accuracy, cls_len, 2)
 		} else {
-			generateExpandableDefinition(c, model, target_accuracy, cls_len, 1)
+			generateExpandableDefinition(c, model, target_accuracy, cls_len, 2)
 		}
 	} else if number_of_models == 3 {
 		for i := 0; i < number_of_models; i++ {
@@ -1550,7 +1549,7 @@ func generateExpandableDefinitions(c BasePack, model *BaseModel, target_accuracy
 	} else {
 		// TODO handle incrisea the complexity
 		for i := 0; i < number_of_models; i++ {
-			generateExpandableDefinition(c, model, target_accuracy, cls_len, 1)
+			generateExpandableDefinition(c, model, target_accuracy, cls_len, 2)
 		}
 	}
 
@@ -1702,21 +1701,39 @@ func RunTaskRetrain(b BasePack, task Task) (err error) {
 
 	task.UpdateStatusLog(b, TASK_RUNNING, "Model retraining")
 
-	defId, err := GetDbVar[string](db, "md.id", "models as m inner join model_definition as md on m.id = md.model_id where m.id=$1;", task.ModelId)
+	var defData struct {
+		Id             string  `db:"md.id"`
+		TargetAcuuracy float64 `db:"md.target_accuracy"`
+	}
+
+	err = GetDBOnce(db, &defData, "models as m inner join model_definition as md on m.id = md.model_id where m.id=$1;", task.ModelId)
 	if err != nil {
 		failed()
 		return
 	}
 
-	// This is something I have to check
-	acc, err := trainDefinitionExpandExp(b, model, *defId, false)
-	if err != nil {
+	var acc float64 = 0
+	var epocs = 0
+	// TODO make max epochs come from db
+	for acc*100 < defData.TargetAcuuracy && epocs < 20 {
+		// This is something I have to check
+		acc, err = trainDefinitionExpandExp(b, model, defData.Id, epocs > 0)
+		if err != nil {
+			failed()
+			return
+		}
+
+		l.Info("Retrained model", "accuracy", acc, "target", defData.TargetAcuuracy)
+
+		epocs += 1
+	}
+
+	if acc*100 < defData.TargetAcuuracy {
+		l.Error("Model never achived targetd accuracy", "acc", acc*100, "target", defData.TargetAcuuracy)
 		failed()
 		return
 	}
 
-	l.Info("Retrained model", "accuracy", acc)
-
 	// TODO check accuracy
 	err = UpdateStatus(db, "models", model.Id, READY)
 	if err != nil {
diff --git a/logic/stats/index.go b/logic/stats/index.go
index 48f14a6..648d37d 100644
--- a/logic/stats/index.go
+++ b/logic/stats/index.go
@@ -6,4 +6,5 @@ import (
 
 func HandleStats(handle *Handle) {
 	HandlePublicStats(handle)
+	handleTasksStats(handle)
 }
diff --git a/logic/stats/tasks.go b/logic/stats/tasks.go
new file mode 100644
index 0000000..c6fcb72
--- /dev/null
+++ b/logic/stats/tasks.go
@@ -0,0 +1,95 @@
+package stats
+
+import (
+	"time"
+
+	. "git.andr3h3nriqu3s.com/andr3/fyp/logic/db_types"
+	. "git.andr3h3nriqu3s.com/andr3/fyp/logic/tasks/utils"
+	. "git.andr3h3nriqu3s.com/andr3/fyp/logic/utils"
+)
+
+func handleTasksStats(handle *Handle) {
+	type ModelTasksStatsRequest struct {
+		ModelId string `json:"model_id" validate:"required"`
+	}
+	PostAuthJson(handle, "/stats/task/model/day", User_Normal, func(c *Context, dat *ModelTasksStatsRequest) *Error {
+		model, err := GetBaseModel(c, dat.ModelId)
+		if err == ModelNotFoundError {
+			return c.JsonBadRequest("Model not found!")
+		} else if err != nil {
+			return c.E500M("Failed to get model", err)
+		}
+
+		now := time.Now()
+		now = time.Date(now.Year(), now.Month(), now.Day(), 0, 0, 0, 0, now.Location())
+
+		tasks, err := GetDbMultitple[Task](c, "tasks where model_id=$1 and created_on > $2 order by created_on asc;", model.Id, now)
+		if err != nil {
+			return c.E500M("Failed to get task informations", err)
+		}
+
+		type DataPoint struct {
+			Non_classfication_tasks_su  int `json:"nc_success"`
+			Non_classfication_tasks_err int `json:"nc_error"`
+
+			Classfication_su      int `json:"c_success"`
+			Classfication_fal     int `json:"c_failure"`
+			Classfication_un      int `json:"c_unknown"`
+			Classfication_err     int `json:"c_error"`
+			Classfication_pre     int `json:"c_pre_running"`
+			Classfication_running int `json:"c_running"`
+		}
+
+		total := DataPoint{}
+		hours := make([]DataPoint, 24)
+
+		for i := 0; i < 24; i++ {
+			hours[i] = DataPoint{}
+		}
+
+		for i := range tasks {
+			task := tasks[i]
+			hour := task.CreatedOn.Hour()
+			if task.TaskType == int(TASK_TYPE_CLASSIFICATION) {
+				if task.Status == 4 {
+					if task.UserConfirmed == 1 {
+						total.Classfication_su += 1
+						hours[hour].Classfication_su += 1
+					} else if task.UserConfirmed == -1 {
+						total.Classfication_fal += 1
+						hours[hour].Classfication_fal += 1
+					} else {
+						total.Classfication_un += 1
+						hours[hour].Classfication_un += 1
+					}
+				} else if task.Status < 0 {
+					total.Classfication_err += 1
+					hours[hour].Classfication_err += 1
+				} else if task.Status < 2 {
+					total.Classfication_pre += 1
+					hours[hour].Classfication_pre += 1
+				} else if task.Status < 4 {
+					total.Classfication_running += 1
+					hours[hour].Classfication_running += 1
+				}
+			} else {
+				if task.Status >= 0 {
+					total.Non_classfication_tasks_su += 1
+					hours[hour].Non_classfication_tasks_su += 1
+				} else {
+					total.Non_classfication_tasks_err += 1
+					hours[hour].Non_classfication_tasks_err += 1
+				}
+			}
+		}
+
+		data := struct {
+			Total DataPoint   `json:"total"`
+			Hours []DataPoint `json:"hours"`
+		}{
+			Total: total,
+			Hours: hours,
+		}
+		return c.SendJSON(data)
+	})
+}
diff --git a/webpage/src/lib/KeepPageSize.svelte b/webpage/src/lib/KeepPageSize.svelte
deleted file mode 100644
index 9260b21..0000000
--- a/webpage/src/lib/KeepPageSize.svelte
+++ /dev/null
@@ -1,11 +0,0 @@
-<script lang="ts">
-	function updateSize() {
-		let s = document.body.getBoundingClientRect();
-		document.body.style.minHeight = `${Math.max(window.innerHeight, s.height)}px`;
-	}
-	$effect(() => {
-		updateSize();
-	});
-</script>
-
-<svelte:window on:scroll={updateSize} />
diff --git a/webpage/src/lib/Tooltip.svelte b/webpage/src/lib/Tooltip.svelte
index 931fa7f..a32084c 100644
--- a/webpage/src/lib/Tooltip.svelte
+++ b/webpage/src/lib/Tooltip.svelte
@@ -5,20 +5,20 @@
 	let x = $state(0);
 	let y = $state(0);
 
-    let div: HTMLDivElement;
+	let div: HTMLDivElement;
 
 	function mouseOver(event: MouseEvent) {
 		isHovered = true;
 		x = event.pageX + 5;
 		y = event.pageY + 5;
 	}
-    function focus() {
-        isHovered = true;
+	function focus() {
+		isHovered = true;
 
-        const box = div.getBoundingClientRect();
-        x = box.x + 5;
-        y = box.y;
-    }
+		const box = div.getBoundingClientRect();
+		x = box.x + 5;
+		y = box.y;
+	}
 	function mouseMove(event: MouseEvent) {
 		x = event.pageX + 5;
 		y = event.pageY + 5;
@@ -28,7 +28,15 @@
 	}
 </script>
 
-<div bind:this={div} on:mouseover={mouseOver} on:mouseleave={mouseLeave} on:mousemove={mouseMove} on:focus={focus} role="tooltip">
+<div
+	bind:this={div}
+	on:mouseover={mouseOver}
+	on:mouseleave={mouseLeave}
+	on:mousemove={mouseMove}
+	on:focus={focus}
+	role="tooltip"
+	class="tooltipContainer"
+>
 	<slot />
 </div>
 
@@ -37,6 +45,9 @@
 {/if}
 
 <style>
+	.tooltipContainer {
+		display: inline-block;
+	}
 	.tooltip {
 		border: 1px solid #ddd;
 		box-shadow: 1px 1px 1px #ddd;
diff --git a/webpage/src/routes/models/edit/+page.svelte b/webpage/src/routes/models/edit/+page.svelte
index f3681f4..d3af0c9 100644
--- a/webpage/src/routes/models/edit/+page.svelte
+++ b/webpage/src/routes/models/edit/+page.svelte
@@ -42,7 +42,6 @@
 	import ModelDataPage from './ModelDataPage.svelte';
 
 	import 'src/styles/forms.css';
-	import KeepPageSize from 'src/lib/KeepPageSize.svelte';
 
 	let model: Promise<Model> = $state(new Promise(() => {}));
 	let _model: Model | undefined = $state(undefined);
@@ -137,8 +136,6 @@
 	{/await}
 </svelte:head>
 
-<KeepPageSize />
-
 <main>
 	{#if [4, 7].includes(_model?.status ?? 0)}
 		<div class="big-msg warning">The Model is currently training</div>
diff --git a/webpage/src/routes/models/edit/DeleteModel.svelte b/webpage/src/routes/models/edit/DeleteModel.svelte
index 333e3cf..01e0bcd 100644
--- a/webpage/src/routes/models/edit/DeleteModel.svelte
+++ b/webpage/src/routes/models/edit/DeleteModel.svelte
@@ -27,7 +27,7 @@
 	}
 </script>
 
-<form on:submit|preventDefault={deleteModel} class:submmited>
+<form on:submit|preventDefault={deleteModel} class:submmited class="danger-bg">
 	<fieldset>
 		<label for="name">
 			To delete this model please type "{model.name}":
diff --git a/webpage/src/routes/models/edit/ModelDataPage.svelte b/webpage/src/routes/models/edit/ModelDataPage.svelte
index 6490be9..6bfb64c 100644
--- a/webpage/src/routes/models/edit/ModelDataPage.svelte
+++ b/webpage/src/routes/models/edit/ModelDataPage.svelte
@@ -2,12 +2,12 @@
 	import { createEventDispatcher } from 'svelte';
 	import type { Model } from './+page.svelte';
 	import ModelData from './ModelData.svelte';
-	import { post } from 'src/lib/requests.svelte';
+	import { post, showMessage } from 'src/lib/requests.svelte';
 	import ModelDataPageStatsGraph from './ModelDataPageStatsGraph.svelte';
 	import type { ModelStats } from './types';
 	import DeleteZip from './DeleteZip.svelte';
 
-	let { model, active } = $props<{ model: Model; active?: boolean }>();
+	let { model, active }: { model: Model; active?: boolean } = $props();
 
 	const dispatch = createEventDispatcher<{ reload: void }>();
 
@@ -21,26 +21,28 @@
 		if (!model) return;
 		try {
 			stats = await post(`models/class/stats`, { id: model.id });
-		} catch {
-			return;
+		} catch (e) {
+			showMessage(e);
 		}
 	}
 </script>
 
-<div class="content" class:selected={active}>
-	{#if stats}
-		<ModelDataPageStatsGraph data={stats} />
-	{/if}
+{#if active}
+	<div class="content selected">
+		{#if stats}
+			<ModelDataPageStatsGraph data={stats} />
+		{/if}
 
-	{#if [-6, -2].includes(model.status)}
-		<DeleteZip {model} on:reload={() => dispatch('reload')} expand />
-	{/if}
+		{#if [-6, -2].includes(model.status)}
+			<DeleteZip {model} on:reload={() => dispatch('reload')} expand />
+		{/if}
 
-	<ModelData
-		{model}
-		on:reload={() => {
-			getData();
-			dispatch('reload');
-		}}
-	/>
-</div>
+		<ModelData
+			{model}
+			on:reload={() => {
+				getData();
+				dispatch('reload');
+			}}
+		/>
+	</div>
+{/if}
diff --git a/webpage/src/routes/models/edit/TasksDataPage.svelte b/webpage/src/routes/models/edit/TasksDataPage.svelte
index 4d44dd9..204abc6 100644
--- a/webpage/src/routes/models/edit/TasksDataPage.svelte
+++ b/webpage/src/routes/models/edit/TasksDataPage.svelte
@@ -2,14 +2,18 @@
 	import { post } from 'src/lib/requests.svelte';
 	import type { Model } from 'src/routes/models/edit/+page.svelte';
 	import RunModel from './RunModel.svelte';
-	import TasksTable from './TasksTable.svelte';
+	import TasksTable from './tasks/TasksTable.svelte';
+	import Stats from './tasks/Stats.svelte';
 
-	const { active, model } = $props<{ active?: boolean; model: Model }>();
+	const { active, model }: { active?: boolean; model: Model } = $props();
 
 	let table: TasksTable;
 </script>
 
-<div class="content" class:selected={active}>
-	<RunModel {model} on:upload={() => table.getList()} on:taskReload={() => table.getList()} />
-	<TasksTable {model} bind:this={table} />
-</div>
+{#if active}
+	<div class="content selected">
+		<RunModel {model} on:upload={() => table.getList()} on:taskReload={() => table.getList()} />
+		<TasksTable {model} bind:this={table} />
+		<Stats {model} />
+	</div>
+{/if}
diff --git a/webpage/src/routes/models/edit/tasks/Stats.svelte b/webpage/src/routes/models/edit/tasks/Stats.svelte
new file mode 100644
index 0000000..9d25b1a
--- /dev/null
+++ b/webpage/src/routes/models/edit/tasks/Stats.svelte
@@ -0,0 +1,163 @@
+<script lang="ts">
+	import { onDestroy, onMount } from 'svelte';
+	import type { Model } from '../+page.svelte';
+	import { post, showMessage } from 'src/lib/requests.svelte';
+	import type { DataPoint, TasksStatsDay } from 'src/types/stats/task';
+
+	import {
+		Chart,
+		Title,
+		Tooltip,
+		Legend,
+		CategoryScale,
+		LinearScale,
+		PieController,
+		LineController,
+		Colors,
+		ArcElement,
+		PointElement,
+		LineElement
+	} from 'chart.js';
+	import ModelData from '../ModelData.svelte';
+
+	Chart.register(
+		Title,
+		Tooltip,
+		Legend,
+		ArcElement,
+		CategoryScale,
+		LinearScale,
+		PieController,
+		LineController,
+		Colors,
+		PointElement,
+		LineElement
+	);
+
+	let { model }: { model: Model } = $props();
+	// Load data
+	$effect(() => {
+		getStats();
+	});
+
+	async function getStats() {
+		if (pieChart && lineChart) {
+			return;
+		}
+		try {
+			const s = await post('stats/task/model/day', {
+				model_id: model.id
+			});
+			console.log(s);
+			createPie(s);
+			createLine(s);
+		} catch (e) {
+			showMessage(e);
+		}
+	}
+
+	let pie: HTMLCanvasElement;
+	let pieChart: Chart<'pie'> | undefined;
+	function createPie(s: TasksStatsDay) {
+		if (pieChart) {
+			pieChart.destroy();
+		}
+		let t = s.total;
+		pieChart = new Chart(pie, {
+			type: 'pie',
+			data: {
+				labels: [
+					'Classfication Error',
+					'Classfication Success',
+					'Classfication Failure',
+					'Classfication Preparing',
+					'Classfication Running',
+					'Classfication Unknown',
+					'Non Classfication Error',
+					'Non Classfication Success'
+				],
+				datasets: [
+					{
+						label: 'Total',
+						data: [
+							t.c_error,
+							t.c_success,
+							t.c_failure,
+							t.c_pre_running,
+							t.c_running,
+							t.c_unknown,
+							t.nc_error,
+							t.nc_success
+						]
+					}
+				]
+			},
+			options: {
+				animation: false
+			}
+		});
+	}
+
+	let line: HTMLCanvasElement;
+	let lineChart: Chart<'line'> | undefined;
+	function createLine(s: TasksStatsDay) {
+		if (lineChart) {
+			lineChart.destroy();
+		}
+		const keys: (keyof DataPoint)[] = Object.keys(s.total) as any;
+		let names: Record<keyof DataPoint, string> = {
+			c_error: 'Classfication Error',
+			c_success: 'Classfication Success',
+			c_failure: 'Classfication Failure',
+			c_pre_running: 'Classfication Preparing',
+			c_running: 'Classfication Running',
+			c_unknown: 'Classfication Unknown',
+			nc_error: 'Non Classfication Error',
+			nc_success: 'Non Classfication Success'
+		};
+		let t = s.total;
+		let labels = new Array(24).fill(0).map((_, i) => i);
+		lineChart = new Chart(line, {
+			type: 'line',
+			data: {
+				labels: labels,
+				datasets: keys.map((key) => ({
+					label: names[key],
+					data: s.hours.map((e) => e[key])
+				}))
+			},
+			options: {
+				animation: false
+			}
+		});
+	}
+
+	onDestroy(() => {
+		if (pieChart) {
+			pieChart.destroy();
+			pieChart = undefined;
+		}
+		if (lineChart) {
+			lineChart.destroy();
+			lineChart = undefined;
+		}
+	});
+</script>
+
+<h1>Statistics (Day)</h1>
+
+<h2>Total</h2>
+<div>
+	<canvas bind:this={pie}></canvas>
+</div>
+
+<h2>Hourly</h2>
+<div>
+	<canvas bind:this={line}></canvas>
+</div>
+
+<style lang="scss">
+	canvas {
+		width: 100%;
+	}
+</style>
diff --git a/webpage/src/routes/models/edit/TasksTable.svelte b/webpage/src/routes/models/edit/tasks/TasksTable.svelte
similarity index 75%
rename from webpage/src/routes/models/edit/TasksTable.svelte
rename to webpage/src/routes/models/edit/tasks/TasksTable.svelte
index 666e8b6..c5029d5 100644
--- a/webpage/src/routes/models/edit/TasksTable.svelte
+++ b/webpage/src/routes/models/edit/tasks/TasksTable.svelte
@@ -29,12 +29,31 @@
 		3: 'Task running',
 		4: 'Task complete'
 	};
+	export const TaskTypeIcons: Record<number, string> = {
+		[-2]: 'bi-exclamation-octagon-fill',
+		[-1]: 'bi-exclamation-diamond-fill',
+		0: 'database-fill-gear',
+		1: 'database-fill-up',
+		2: 'database-fill-lock',
+		3: 'database-fill-gear',
+		4: 'database-fill-check'
+	};
+	export const TaskTypeColors: Record<number, string> = {
+		[-2]: 'red',
+		[-1]: 'red',
+		0: 'blue',
+		1: 'blue',
+		2: 'orange',
+		3: 'orange',
+		4: 'green'
+	};
 </script>
 
 <script lang="ts">
 	import { post, showMessage } from 'src/lib/requests.svelte';
-	import type { Model } from './+page.svelte';
+	import type { Model } from '../+page.svelte';
 	import MessageSimple from 'src/lib/MessageSimple.svelte';
+	import Tooltip from 'src/lib/Tooltip.svelte';
 
 	let { model }: { model: Model } = $props();
 
@@ -45,7 +64,7 @@
 	export async function getList() {
 		try {
 			const res = await post('tasks/list', {
-				id: model.id,
+				model_id: model.id,
 				page: page
 			});
 			showNext = res.show_next;
@@ -134,16 +153,22 @@
 								{:else}
 									TODO {task.user_confirmed}
 								{/if}
-								{#if task.user_confirmed == -1}
-									<button type="button" on:click={userPreception(task.id, 1)}>
-										<span class="bi bi-check"></span>
-									</button>
-								{/if}
-								{#if task.user_confirmed == 1}
-									<button class="danger" type="button" on:click={userPreception(task.id, -1)}>
-										<span class="bi bi-x-lg"></span>
-									</button>
-								{/if}
+								<div>
+									{#if task.user_confirmed != 1}
+										<Tooltip title="Agree with the result of the task">
+											<button type="button" on:click={userPreception(task.id, 1)}>
+												<span class="bi bi-check"></span>
+											</button>
+										</Tooltip>
+									{/if}
+									{#if task.user_confirmed != -1}
+										<Tooltip title="Disagree with the result">
+											<button class="danger" type="button" on:click={userPreception(task.id, -1)}>
+												<span class="bi bi-x-lg"></span>
+											</button>
+										</Tooltip>
+									{/if}
+								</div>
 							{:else}
 								-
 							{/if}
@@ -155,14 +180,20 @@
 						{#if task.status == 4}
 							{#if task.type == 1}
 								{@const temp = JSON.parse(task.result)}
-								{temp.class}({temp.confidence * 100}%)
+								{temp.class}({Math.round(temp.confidence * 100000) / 1000}%)
 							{:else}
 								{task.result}
 							{/if}
 						{/if}
 					</td>
-					<td>
-						{TaskTypeStrings[task.status]}
+					<td style="text-align: center;">
+						<Tooltip title={TaskTypeStrings[task.status]}>
+							<span
+								class="bi bi-{TaskTypeIcons[task.status]}"
+								style="color: {TaskTypeColors[task.status]}; font-size: 1.5em;"
+							>
+							</span>
+						</Tooltip>
 					</td>
 					<td>
 						{task.status_message}
diff --git a/webpage/src/types/stats/task.ts b/webpage/src/types/stats/task.ts
new file mode 100644
index 0000000..98072d5
--- /dev/null
+++ b/webpage/src/types/stats/task.ts
@@ -0,0 +1,15 @@
+export type DataPoint = {
+	nc_success: number;
+	nc_error: number;
+	c_success: number;
+	c_failure: number;
+	c_unknown: number;
+	c_error: number;
+	c_pre_running: number;
+	c_running: number;
+};
+
+export type TasksStatsDay = {
+	total: DataPoint;
+	hours: DataPoint[];
+};