Added start stop button. Make test pass.