Adjusting code for Live site