Update for button

Signed-off-by: Ben Merrell <ben@digitalboutique.co.uk>