Update to onload js

Try to make it work