Adding jquery script to add missing attribute