Adjusting the stopbutton, using enum instead of multiple classes