Update to include button pins