Edit Mode Button Enables and Disables Correctly