Add inspection on return stmt