diff --git a/comp90015_assignment2.iml b/comp90015_assignment2.iml
new file mode 100644
index 0000000000000000000000000000000000000000..c90834f2d607afe55e6104d8aa2cdfffb713f688
--- /dev/null
+++ b/comp90015_assignment2.iml
@@ -0,0 +1,11 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="JAVA_MODULE" version="4">
+  <component name="NewModuleRootManager" inherit-compiler-output="true">
+    <exclude-output />
+    <content url="file://$MODULE_DIR$">
+      <sourceFolder url="file://$MODULE_DIR$/src" isTestSource="false" />
+    </content>
+    <orderEntry type="inheritedJdk" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+</module>
\ No newline at end of file
diff --git a/LICENSE b/src/LICENSE
similarity index 100%
rename from LICENSE
rename to src/LICENSE
diff --git a/DrawingArea.java b/src/client/DrawingArea.java
similarity index 99%
rename from DrawingArea.java
rename to src/client/DrawingArea.java
index 0c2c1dc2dd30092c987fd54e73603f3283aedcb5..4bd67e64820af78685bc9cc8a1a7639c4fcb8030 100644
--- a/DrawingArea.java
+++ b/src/client/DrawingArea.java
@@ -1,3 +1,5 @@
+package client;
+
 import javax.imageio.ImageIO;
 import javax.swing.*;
 import java.awt.*;
diff --git a/PaintGUI.java b/src/client/PaintGUI.java
similarity index 99%
rename from PaintGUI.java
rename to src/client/PaintGUI.java
index df27b3bdebd88e706e0e4cae0569bc77385b8e47..02ea1bce4399175d6903809aea3653751cd2046e 100644
--- a/PaintGUI.java
+++ b/src/client/PaintGUI.java
@@ -1,3 +1,5 @@
+package client;
+
 import javax.swing.*;
 import java.awt.*;
 import java.awt.event.ActionEvent;