Address minor JS review comments