diff --git a/Examples/Nodify.Calculator/CreateModelDialog.xaml b/Examples/Nodify.Calculator/CreateModelDialog.xaml
index e295846..cc35c55 100644
--- a/Examples/Nodify.Calculator/CreateModelDialog.xaml
+++ b/Examples/Nodify.Calculator/CreateModelDialog.xaml
@@ -33,9 +33,42 @@
                   Background="#3E3E42" Foreground="White" RowBackground="#3E3E42"
                   AlternatingRowBackground="#333337" GridLinesVisibility="None"
                   HeadersVisibility="Column" BorderBrush="#555">
+            <DataGrid.ColumnHeaderStyle>
+                <Style TargetType="DataGridColumnHeader">
+                    <Setter Property="Background" Value="#2D2D30" />
+                    <Setter Property="Foreground" Value="White" />
+                    <Setter Property="Padding" Value="6 4" />
+                    <Setter Property="BorderBrush" Value="#555" />
+                    <Setter Property="BorderThickness" Value="0 0 1 1" />
+                </Style>
+            </DataGrid.ColumnHeaderStyle>
             <DataGrid.Columns>
-                <DataGridTextColumn Header="Name" Binding="{Binding Name}" Width="*" />
+                <DataGridTextColumn Header="Name" Binding="{Binding Name}" Width="*">
+                    <DataGridTextColumn.ElementStyle>
+                        <Style TargetType="TextBlock">
+                            <Setter Property="Foreground" Value="White" />
+                        </Style>
+                    </DataGridTextColumn.ElementStyle>
+                    <DataGridTextColumn.EditingElementStyle>
+                        <Style TargetType="TextBox">
+                            <Setter Property="Foreground" Value="Black" />
+                            <Setter Property="Background" Value="White" />
+                        </Style>
+                    </DataGridTextColumn.EditingElementStyle>
+                </DataGridTextColumn>
                 <DataGridComboBoxColumn Header="Type" SelectedItemBinding="{Binding Type}" Width="120">
+                    <DataGridComboBoxColumn.ElementStyle>
+                        <Style TargetType="ComboBox">
+                            <Setter Property="Foreground" Value="White" />
+                            <Setter Property="Background" Value="#3E3E42" />
+                        </Style>
+                    </DataGridComboBoxColumn.ElementStyle>
+                    <DataGridComboBoxColumn.EditingElementStyle>
+                        <Style TargetType="ComboBox">
+                            <Setter Property="Foreground" Value="Black" />
+                            <Setter Property="Background" Value="White" />
+                        </Style>
+                    </DataGridComboBoxColumn.EditingElementStyle>
                     <DataGridComboBoxColumn.ItemsSource>
                         <x:Array Type="sys:String" xmlns:sys="clr-namespace:System;assembly=System.Runtime">
                             <sys:String>string</sys:String>