diff --git a/plugins/advancedviewplugin/advancedviewplugin.pro b/plugins/advancedviewplugin/advancedviewplugin.pro
index 348a6bc..7e2e846 100644
--- a/plugins/advancedviewplugin/advancedviewplugin.pro
+++ b/plugins/advancedviewplugin/advancedviewplugin.pro
@@ -16,11 +16,15 @@ DEFINES += QT_DEPRECATED_WARNINGS QT_DISABLE_DEPRECATED_BEFORE=0x060000 QT_MESSA
 
 HEADERS += advancedviewplugin.h \
            advancedviewwidget.h \
-           advanvedviewdialog.h
+           advanvedviewdialog.h \
+           views/bookingsview.h \
+           views/timeassignmentsview.h
 
 SOURCES += advancedviewplugin.cpp \
            advancedviewwidget.cpp \
-           advanvedviewdialog.cpp
+           advanvedviewdialog.cpp \
+           views/bookingsview.cpp \
+           views/timeassignmentsview.cpp
 
 FORMS += advanvedviewdialog.ui
 
diff --git a/plugins/advancedviewplugin/advanvedviewdialog.ui b/plugins/advancedviewplugin/advanvedviewdialog.ui
index bf1934a..da2fe7c 100644
--- a/plugins/advancedviewplugin/advanvedviewdialog.ui
+++ b/plugins/advancedviewplugin/advanvedviewdialog.ui
@@ -19,8 +19,8 @@
      <property name="orientation">
       <enum>Qt::Vertical</enum>
      </property>
-     <widget class="QTreeView" name="bookingsView"/>
-     <widget class="QTreeView" name="timeAssignmentsView"/>
+     <widget class="BookingsView" name="bookingsView"/>
+     <widget class="TimeAssignmentsView" name="timeAssignmentsView"/>
     </widget>
    </item>
    <item>
@@ -35,6 +35,18 @@
    </item>
   </layout>
  </widget>
+ <customwidgets>
+  <customwidget>
+   <class>BookingsView</class>
+   <extends>QTreeView</extends>
+   <header>views/bookingsview.h</header>
+  </customwidget>
+  <customwidget>
+   <class>TimeAssignmentsView</class>
+   <extends>QTreeView</extends>
+   <header>views/timeassignmentsview.h</header>
+  </customwidget>
+ </customwidgets>
  <resources/>
  <connections>
   <connection>
diff --git a/plugins/advancedviewplugin/views/bookingsview.cpp b/plugins/advancedviewplugin/views/bookingsview.cpp
new file mode 100644
index 0000000..07e5b0f
--- /dev/null
+++ b/plugins/advancedviewplugin/views/bookingsview.cpp
@@ -0,0 +1,7 @@
+#include "bookingsview.h"
+
+BookingsView::BookingsView(QWidget *parent) :
+    QTreeView(parent)
+{
+
+}
diff --git a/plugins/advancedviewplugin/views/bookingsview.h b/plugins/advancedviewplugin/views/bookingsview.h
new file mode 100644
index 0000000..97966f3
--- /dev/null
+++ b/plugins/advancedviewplugin/views/bookingsview.h
@@ -0,0 +1,14 @@
+#ifndef BOOKINGSVIEW_H
+#define BOOKINGSVIEW_H
+
+#include <QTreeView>
+
+class BookingsView : public QTreeView
+{
+    Q_OBJECT
+
+public:
+    explicit BookingsView(QWidget *parent = Q_NULLPTR);
+};
+
+#endif // BOOKINGSVIEW_H
diff --git a/plugins/advancedviewplugin/views/timeassignmentsview.cpp b/plugins/advancedviewplugin/views/timeassignmentsview.cpp
new file mode 100644
index 0000000..4d862c5
--- /dev/null
+++ b/plugins/advancedviewplugin/views/timeassignmentsview.cpp
@@ -0,0 +1,7 @@
+#include "timeassignmentsview.h"
+
+TimeAssignmentsView::TimeAssignmentsView(QWidget *parent) :
+    QTreeView(parent)
+{
+
+}
diff --git a/plugins/advancedviewplugin/views/timeassignmentsview.h b/plugins/advancedviewplugin/views/timeassignmentsview.h
new file mode 100644
index 0000000..3064a33
--- /dev/null
+++ b/plugins/advancedviewplugin/views/timeassignmentsview.h
@@ -0,0 +1,14 @@
+#ifndef TIMEASSIGNMENTSVIEW_H
+#define TIMEASSIGNMENTSVIEW_H
+
+#include <QTreeView>
+
+class TimeAssignmentsView : public QTreeView
+{
+    Q_OBJECT
+
+public:
+    explicit TimeAssignmentsView(QWidget *parent = 0);
+};
+
+#endif // TIMEASSIGNMENTSVIEW_H