Modify js code