Making small adjustments to the CSS and cleaning up the JS