Added button and changed sizings