Updates for new buttons etc