Adding new JS to help display content correct