'Solution'

app/main.py

@@ -1,3 +1,28 @@
+import os
+from typing import Optional, Type
+
+
 class CleanUpFile:
-    # write your code here
-    pass
+    def __init__(self, filename: str) -> None:
+        self.filename = filename
+
+    def __enter__(self) -> "CleanUpFile":
+        return self
+
+    def __exit__(
+        self,
+        exc_type: Optional[Type[BaseException]],
+        exc_value: Optional[BaseException],
+        traceback: Optional[object]
+    ) -> None:
+        if os.path.exists(self.filename):
+            os.remove(self.filename)
+            print(f"File '{self.filename}' has been removed.")
+        else:
+            print(f"File '{self.filename}' does not exist.")
+
+
+# Example usage:
+with CleanUpFile("file.txt"):
+    with open("file.txt", "w") as file:
+        file.write("Hello Mate!")