Update script for ajax compatibility