Bug: fixed freak outs with Generate Button