remove unused button and change the base model