Add setEnabled method