diff --git a/.idea/vcs.xml b/.idea/vcs.xml
index 6c0b863..288b36b 100644
--- a/.idea/vcs.xml
+++ b/.idea/vcs.xml
@@ -2,5 +2,6 @@
 <project version="4">
   <component name="VcsDirectoryMappings">
     <mapping directory="$PROJECT_DIR$/.." vcs="Git" />
+    <mapping directory="$PROJECT_DIR$" vcs="Git" />
   </component>
 </project>
\ No newline at end of file
diff --git a/app/src/main/java/com/example/androidProjectOnTFLite/CapturePhotoFragment.kt b/app/src/main/java/com/example/androidProjectOnTFLite/CapturePhotoFragment.kt
index d72bb76..2deb768 100644
--- a/app/src/main/java/com/example/androidProjectOnTFLite/CapturePhotoFragment.kt
+++ b/app/src/main/java/com/example/androidProjectOnTFLite/CapturePhotoFragment.kt
@@ -11,24 +11,29 @@ import android.view.View
 import android.view.ViewGroup
 import android.widget.Button
 import android.widget.ImageView
+import android.widget.TextView
 import androidx.fragment.app.Fragment
 import kotlinx.android.synthetic.main.fragment_capture_photo.*
 import java.io.ByteArrayOutputStream
 
 
 class CapturePhotoFragment : Fragment() {
-    lateinit var Camera: ImageView
-    lateinit var Button: Button
-    lateinit var intent:Intent
-
-    override fun onCreateView(
-        inflater: LayoutInflater, container: ViewGroup?,
-        savedInstanceState: Bundle?
-    ): View? {
+    private lateinit var camera: ImageView
+    private lateinit var btnCapturePhoto: Button
+    private lateinit var intent:Intent
+    private lateinit var textClass:TextView
+    override fun onCreate(savedInstanceState: Bundle?) {
+        super.onCreate(savedInstanceState)
         intent = Intent(MediaStore.ACTION_IMAGE_CAPTURE)
         startActivityForResult(
             intent,
             CAPTURE_IMAGE_ACTIVITY_REQUEST_CODE)
+    }
+    override fun onCreateView(
+        inflater: LayoutInflater, container: ViewGroup?,
+        savedInstanceState: Bundle?
+    ): View? {
+
         // Inflate the layout for this fragment
         return inflater.inflate(R.layout.fragment_capture_photo, container, false)
     }
@@ -36,10 +41,10 @@ class CapturePhotoFragment : Fragment() {
     override fun onViewCreated(view: View, savedInstanceState: Bundle?) {
 
 
-        Button = btnCapturePhoto
-        Camera = imgViewCamera
-
-        Button.setOnClickListener {
+        btnCapturePhoto = btn_capture_photo
+        camera = img_view_camera
+        textClass = text_view_class
+        btnCapturePhoto.setOnClickListener {
                 startActivityForResult(
                     intent,
                     CAPTURE_IMAGE_ACTIVITY_REQUEST_CODE)
@@ -63,7 +68,11 @@ class CapturePhotoFragment : Fragment() {
                     byteArray, 0,
                     byteArray.size
                 )
-                Camera.setImageBitmap(bitmap)
+                camera.setImageBitmap(bitmap)
+                textClass.text = "Loading..."
+                // do classification
+                textClass.text = "Class"
+                btnCapturePhoto.visibility=View.VISIBLE
             }
         }
     }
diff --git a/app/src/main/res/layout/fragment_capture_photo.xml b/app/src/main/res/layout/fragment_capture_photo.xml
index c342bc1..8fecd5b 100644
--- a/app/src/main/res/layout/fragment_capture_photo.xml
+++ b/app/src/main/res/layout/fragment_capture_photo.xml
@@ -1,21 +1,36 @@
 <?xml version="1.0" encoding="utf-8"?>
 <LinearLayout xmlns:android="http://schemas.android.com/apk/res/android"
     xmlns:tools="http://schemas.android.com/tools"
-    android:id="@+id/frameLayout2"
+    android:id="@+id/linear_layout"
     android:layout_width="match_parent"
     android:layout_height="match_parent"
     android:orientation="vertical"
     tools:context=".CapturePhotoFragment">
 
+    <TextView
+        android:id="@+id/text_view_class"
+        android:layout_width="match_parent"
+        android:layout_height="wrap_content"
+        android:textAlignment="center" />
+
+    <Space
+        android:layout_width="match_parent"
+        android:layout_height="30dp" />
+
     <ImageView
-        android:id="@+id/imgViewCamera"
+        android:id="@+id/img_view_camera"
         android:layout_width="match_parent"
-        android:layout_height="394dp"
-        android:src="@android:drawable/ic_menu_camera" />
+        android:layout_height="450dp" />
+
+    <Space
+        android:layout_width="match_parent"
+        android:layout_height="30dp" />
 
     <Button
-        android:id="@+id/btnCapturePhoto"
+        android:id="@+id/btn_capture_photo"
         android:layout_width="match_parent"
-        android:layout_height="wrap_content"
-        android:text="Button" />
+        android:layout_height="80dp"
+        android:text="Take Photo"
+        android:visibility="invisible" />
+
 </LinearLayout>
\ No newline at end of file