$ causes a clash in wordpress so you need to

Change the ‘$’ in this:

$(window).on('load'

to this:

jQuery(window).on('load'

Reason: https://mkyong.com/jquery/jquery-is-not-working-in-wordpress-solution/