Updated function to call when button clicked
Updated changes made on Codepen, page now works, just not very pretty