Added new button and changed design a little bit