Cleaned up setEnabled button code