- Added download and restart fragment
- Added some more logic to version details fragment
diff --git a/FairphoneUpdater/src/com/fairphone/updater/fragments/MainFragment.java b/FairphoneUpdater/src/com/fairphone/updater/fragments/MainFragment.java
index e037627..0da0260 100644
--- a/FairphoneUpdater/src/com/fairphone/updater/fragments/MainFragment.java
+++ b/FairphoneUpdater/src/com/fairphone/updater/fragments/MainFragment.java
@@ -24,6 +24,7 @@
 	private TextView mUpdateAvailableNameText;
 	private Button mUpdateAvailableInstallButton;
 	private Button mOtherOSOptionsButton;
+	private Version mDeviceVersion;
 
 	@Override
 	public View onCreateView(LayoutInflater inflater, ViewGroup container,
@@ -33,7 +34,8 @@
 
 		setupLayout(view);
 
-		mainActivity.updateHeader(HeaderType.MAIN, "");
+		mDeviceVersion =  mainActivity.getDeviceVersion();
+		updateHeader();
 		updateCurrentVersionGroup();
 		toogleUpdateAvailableGroup();
 		updateOtherOSOptionsGroup();
@@ -41,6 +43,14 @@
 		return view;
 	}
 
+	private void updateHeader() {
+		if(Version.IMAGE_TYPE_FAIRPHONE.equalsIgnoreCase(mDeviceVersion.getImageType())){
+			mainActivity.updateHeader(HeaderType.MAIN_FAIRPHONE, "");
+		}else if (Version.IMAGE_TYPE_AOSP.equalsIgnoreCase(mDeviceVersion.getImageType())){
+			mainActivity.updateHeader(HeaderType.MAIN_ANDROID, "");
+		}
+	}
+
 	private void setupLayout(View view) {
 		// Current version group
 		mCurrentVersionNameText = (TextView) view