Minor JS cleanup