CSS Styling, Clean up of JS file