added edit and remove buttons, remove is now working