JS cleanup - switch to new document ready syntax