Small changes to refinement. Added back the buttons