"Stop generation" button added