Small JS style fixes