JS code clean-up. Some simple CSS styling.