Add buttons to update state