                 // Set appearance
                 if let done = doneButton {
                     done.setBackgroundImage(nil, for: .normal, barMetrics: .default)
-                    done.setBackgroundImage(nil, for: .normal, barMetrics: .compact)
-                    done.setBackgroundImage(nil, for: .highlighted, barMetrics: .default)
                     done.setBackgroundImage(nil, for: .highlighted, barMetrics: .compact)
-                    done.setTitleTextAttributes([String : AnyObject](), for: .normal)
-                    done.setTitleTextAttributes([String : AnyObject](), for: .highlighted)
+//                    done.setTitleTextAttributes([String : AnyObject](), for: .normal)
+//                    done.setTitleTextAttributes([String : AnyObject](), for: .highlighted)
                     
                     self.navigationItem.rightBarButtonItem = done
                 }
-            }
-            else {
+            } else {
                 // We're not first so show back button
-                if let navi = navigationController,
-                    let previousViewController = navi.viewControllers[navi.viewControllers.count - 2] as? UINavigationController
-                {
+                if let navi = navigationController, let previousViewController = navi.viewControllers[navi.viewControllers.count - 2] as? UINavigationController {
                     let backButtonTitle = previousViewController.navigationItem.backBarButtonItem != nil ?
                         previousViewController.navigationItem.backBarButtonItem!.title :
                         previousViewController.title


    private func initialisation() {
        // Defaults
        if let vcBasedStatusBarAppearance = Bundle.main
            .object(forInfoDictionaryKey: "UIViewControllerBasedStatusBarAppearance") as? Bool {
           isVCBasedStatusBarAppearance = vcBasedStatusBarAppearance
        } else {
            isVCBasedStatusBarAppearance = true

        // Init grid controller
        gridController = GridViewController()
        
        if let gc = gridController, let navBar = navigationController?.navigationBar {
            let bounds = view.bounds
            let naviHeight = navBar.frame.height + UIApplication.shared.statusBarFrame.height
            
