                         for (index, placeID) in list.placeIDs.enumerated() {
                             if placeID == self.place.id {
                                 print("bookmark exists, changing button")
-                                //let button = UIButton.init(type: UIButtonType.custom)
-                                //button.frame = CGRect(x: 0, y: 0, width: 100, height: 100)
-                                //button.setImage(UIImage(named: "bookmarked-2x.png"), for: .normal)
-                                //button.addTarget(self, action: #selector(self.onBookmarkButton(_:)), for: UIControlEvents.touchUpInside)
-                                //self.bookmarkBarButtonItem = UIBarButtonItem(image: #imageLiteral(resourceName: "bookmarked"), style: UIBarButtonItemStyle.plain, target: self, action: #selector(self.onBookmarkButton(_:)))
-                                self.navigationItem.rightBarButtonItems![0] = UIBarButtonItem(image: #imageLiteral(resourceName: "bookmarked"),
-                                                                                              style: UIBarButtonItemStyle.plain,
-                                                                                              target: self,
-                                                                                              action: #selector(self.onBookmarkButton(_:)))
-                                self.navigationItem.rightBarButtonItems![0].tintColor = UIColor.white
-                                //self.navigationItem.rightBarButtonItems![0] = UIBarButtonItem(customView: button)
-                                //self.bookmarkBarButtonItem = UIBarButtonItem.init(barButtonSystemItem: UIBarButtonSystemItem.bookmarks, target: self, action: #selector(self.onBookmarkButton(_:)))
                                 self.isBookmarked = true
                                 self.bookmarkIndex = index
+                                self.updateBookMarkImage()
                             }
                         }
                     }

    }
    
    // MARK: - Helpers
    private func findBookmarks()  {
        DatabaseRequests.shared.fetchCurrentUserLists(success: {
            lists in
            if let lists = lists {

                    _ = SweetAlert().showAlert("Success!", subTitle: "Added \(self.place.name) to your Bookmarks.", style: .success)
                }
            })
            isBookmarked = true
            bookmarkIndex = bList.placeIDs.count - 1
            bookmarksList = bList
            hasBookmarks = true
            self.updateBookMarkImage()
        }
    }
    
