Updated JS to use the class names specified in index.css