Little cleanup js code

Little cleanup js code