diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index df59bfc..ef85c0d 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -21,11 +21,11 @@ repos:
       - id: commitizen-branch
         stages: [push]
   - repo: https://github.com/psf/black
-    rev: 23.11.0
+    rev: 23.12.0
     hooks:
       - id: black
   - repo: https://github.com/pycqa/isort
-    rev: 5.12.0
+    rev: 5.13.1
     hooks:
       - id: isort
   - repo: https://github.com/pycqa/flake8
diff --git a/apps/tasks/views.py b/apps/tasks/views.py
index 0238fd4..b4d5404 100644
--- a/apps/tasks/views.py
+++ b/apps/tasks/views.py
@@ -1,6 +1,8 @@
 import sys
 from io import StringIO
-from typing import TYPE_CHECKING, Any, Dict
+from resource import RLIMIT_AS, getrlimit, setrlimit
+from signal import SIGALRM, alarm, signal
+from typing import TYPE_CHECKING, Any, Dict, Optional
 
 from django.http import HttpRequest, HttpResponse
 from django.shortcuts import redirect
@@ -21,31 +23,56 @@
     FormMixinBase = FormMixin
 
 
+def signal_handler(signum: int, frame: object) -> None:
+    raise TimeoutError("Time limit exceeded")
+
+
 @celery.task(ignore_result=True)
 def handle_submission(code: str, task_id: int, submission_id: int) -> None:
     task = Task._default_manager.get(id=task_id)
     submission = Submission._default_manager.get(id=submission_id)
 
+    if task.memory_limit is not None:
+        _, hard = getrlimit(RLIMIT_AS)
+        setrlimit(RLIMIT_AS, (task.memory_limit, hard))
+
+    if task.time_limit is not None:
+        signal(SIGALRM, signal_handler)
+        alarm(task.time_limit)
+
+    if (output := compile_code(submission, task, code)) is None:
+        return
+
+    check_answer(submission, task, output)
+
+
+def compile_code(
+    submission: Submission, task: Task, code: str
+) -> Optional[str]:
     input_data = StringIO(task.input_file)
 
-    old_stdin = sys.stdin
     sys.stdin = input_data
-
-    old_stdout = sys.stdout
     sys.stdout = stdout = StringIO()
 
     try:
         eval(compile(code, "<string>", "exec"))
+    except MemoryError:  # pragma: no cover
+        submission.status = "MLE"
+        submission.save()
+        return None
+    except TimeoutError:  # pragma: no cover
+        submission.status = "TLE"
+        submission.save()
+        return None
     except Exception:
         submission.status = "RE"
         submission.save()
-        return
-    finally:
-        sys.stdout = old_stdout
-        sys.stdin = old_stdin
+        return None
+
+    return stdout.getvalue()
 
-    output = stdout.getvalue()
 
+def check_answer(submission: Submission, task: Task, output: str) -> None:
     submission.status = "AC" if output == task.output_file else "WA"
     submission.save()
 
@@ -58,7 +85,6 @@ def handle_submission(code: str, task_id: int, submission_id: int) -> None:
         .exclude(id=submission.id)
         .exists()
     )
-
     is_accepted = submission.status == SubmissionStatus.ACCEPTED
 
     if is_accepted and not has_already_scored:
diff --git a/out.txt b/out.txt
new file mode 100644
index 0000000..af5626b
--- /dev/null
+++ b/out.txt
@@ -0,0 +1 @@
+Hello, world!
diff --git a/setup.cfg b/setup.cfg
index 53db72d..6eb0943 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -38,6 +38,7 @@ omit =
 
 exclude_lines =
     if TYPE_CHECKING:
+    pragma: no cover
 
 [mypy]
 strict = true