how to round edges of buttons in xcode code example

Example: ios rounded button

button.layer.cornerRadius = button.frame.size.height / 2
// must be in the viewDidLoad func