Add buttons, fix bugs