Working on fixing the stop button