Here is what I used previously,
var barButtonItem = UIBarButtonItem(image: backImgs, style: UIBarButtonItemStyle.plain, target: self, action: Selector(\"menu
Make a UIBarButtonItem:
let rightButton: UIBarButtonItem = UIBarButtonItem(title: "Done", style: UIBarButtonItemStyle.done, target: self, action: #selector(LocationViewController.doneButtonClicked(_:)))
Add to NavigationItem:
self.navigationItem.rightBarButtonItem = rightButton
Associated function:
func doneButtonClicked(_ button:UIBarButtonItem!){
print("Done clicked")
}