Schema Update & bug fix