Update controller due to DB changes