Added a button, but I can't seem to wait