Made some small adjustment to proto