Added buttton, and changed look