Change onload to DOM ready