Update code with updated db script and code