Add code to js