added the button and finished the assignment