Changed up the logic for enabling/disabling the buttons