Attached generator to button #146

As soon as the function works, close the issue