Added a stop button, still kinda sorta works