Re-added the delay/button press function