updating some of the code so that it won't break anything