make selection buttons on edit screen work