Prepared the code to remove the download button by dividing the JS