Small style and JS changes