added in some buttons for UI