Wrote some button code

not sure if we'll need it though ...