Putting back accidentally deleted code