Tidy up jQuery script
Tidy up jQuery script