Updates Code to work with local