Made some changes to the js and added a modal