Updates UI tests to add stop button behaviours