adding in buttons and staging