Added button function hopefully