Add button and spacing back