WIP -- main buttons works