added a new button and fixed the interface