Setting up buttons (WIP)