refactor UI: buttons instead of menu