diff --git a/.idea/DSZI.iml b/.idea/DSZI.iml
index bb9849c..0267347 100644
--- a/.idea/DSZI.iml
+++ b/.idea/DSZI.iml
@@ -5,7 +5,7 @@
       <excludeFolder url="file://$MODULE_DIR$/Restaurant/Marta/venv" />
       <excludeFolder url="file://$MODULE_DIR$/venv" />
     </content>
-    <orderEntry type="jdk" jdkName="Python 3.7 (Restaurant)" jdkType="Python SDK" />
+    <orderEntry type="jdk" jdkName="Python 3.7" jdkType="Python SDK" />
     <orderEntry type="sourceFolder" forTests="false" />
   </component>
   <component name="TestRunnerService">
diff --git a/.idea/misc.xml b/.idea/misc.xml
index 4f2d1fe..a2e120d 100644
--- a/.idea/misc.xml
+++ b/.idea/misc.xml
@@ -1,4 +1,4 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <project version="4">
-  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.7 (Restaurant)" project-jdk-type="Python SDK" />
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.7" project-jdk-type="Python SDK" />
 </project>
\ No newline at end of file
diff --git a/Restaurant/main.py b/Restaurant/main.py
index 01bdc52..7d3ce2b 100644
--- a/Restaurant/main.py
+++ b/Restaurant/main.py
@@ -130,7 +130,7 @@ class Dish(object):
 
 
 class Plate(object):
-    def __init__(self, dish, img):
+    def __init__(self, dish, img=None):
         self.dishName = dish.dishName
         self.isEmpty = False
         self.preparationTime = dish.preparationTime