Add buttons and simple functionality