Add search for `return null;`