Updated the js to use css classes instead of hardcoding styling